echo "开始拉取镜像"
wget -P /root/stable-diffusion-webui/models/Stable-diffusion/ "http://public.x-gpu.com/f/NEysj/majicMIX-realistic-%E9%BA%A6%E6%A9%98%E5%86%99%E5%AE%9E-v7.safetensors"
wget -P /root/stable-diffusion-webui/models/Stable-diffusion/ http://public.x-gpu.com/f/zB9T4/%E4%B8%87%E8%B1%A1%E7%86%94%E7%82%89-Anything%20V5-Ink-ink.safetensors
git clone https://github.com/lobehub/sd-webui-lobe-theme.git /root/stable-diffusion-webui/extensions/sd-webui-lobe-theme
# 挂载插件
ln -s /root/cloud/stable-diffusion-webui/extensions/sd-webui-prompt-all-in-one /root/stable-diffusion-webui/extensions/sd-webui-prompt-all-in-one
ln -s /root/cloud/stable-diffusion-webui/extensions/sd_web_ui_preset_utils /root/stable-diffusion-webui/extensions/sd_web_ui_preset_utils
ln -s /root/cloud/stable-diffusion-webui/extensions/sd-webui-controlnet /root/stable-diffusion-webui/extensions/sd-webui-controlnet
echo "镜像拉取完成"

screen -S sd -X quit
echo "Stable Diffuison Webui 已停止..."
bash /scripts/start.d/stablediffusion.sh
echo "Stable Diffusion Webui 正在启动..."
while ! curl --output /dev/null --silent --head --fail http://127.0.0.1:80; do
    echo "正在检查启动状态..."
    sleep 10
done
echo "启动成功，请关闭该窗口，打开 Webui"